@inproceedings{paulus2021raoblackwellizing,
title={Rao-Blackwellizing the Straight-Through Gumbel-Softmax Gradient Estimator},
author={Max B Paulus and Chris J. Maddison and Andreas Krause},
booktitle={International Conference on Learning Representations},
year={2021},
url={https://openreview.net/forum?id=Mk6PZtgAgfq}
}